<template xlang="wxml">
  <view class="tki-tree">
    <view :class="{'show':showTree}" class="tki-tree-mask" @tap="_cancel"></view>
    <view :class="{'show':showTree}" class="tki-tree-cnt">
      <view class="tki-tree-bar">
        <view :style="{'color':cancelColor}" class="tki-tree-bar-cancel" hover-class="hover-c" @tap="_cancel">取消
        </view>
        <view :style="{'color':titleColor}" class="tki-tree-bar-title">{{ title }}</view>
        <view :style="{'color':confirmColor}" class="tki-tree-bar-confirm" hover-class="hover-c" @tap="_confirm">确定
        </view>
      </view>
      <view class="tki-tree-view">
        <scroll-view :scroll-y="true" class="tki-tree-view-sc">
          <block v-for="(item, index) in treeList" :key="index">
            <view :class="{
							border: border === true,
							show: item.show,
							last: item.lastRank,
							showchild: item.showChild,
							open: item.open,
						}" :style="[{
							paddingLeft: item.rank*15 + 'px',
							zIndex: item.rank*-1 +50
						}]"
                  class="tki-tree-item">
              <view class="tki-tree-label" @tap.stop="_treeItemTap(item, index)">
                <image :src="item.lastRank ? lastIcon : item.showChild ? currentIcon : defaultIcon"
                       class="tki-tree-icon"></image>
                {{ item.name }}
              </view>
              <view v-if="selectParent?true:item.lastRank" class="tki-tree-check"
                    @tap.stop="_treeItemSelect(item, index)">
                <view v-if="item.checked" :class="{'radio':!multiple}" :style="{'border-color':confirmColor}"
                      class="tki-tree-check-yes">
                  <view :style="{'background-color':confirmColor}" class="tki-tree-check-yes-b"></view>
                </view>
                <view v-else :class="{'radio':!multiple}" :style="{'border-color':confirmColor}"
                      class="tki-tree-check-no"></view>
              </view>
            </view>
          </block>
        </scroll-view>
      </view>
    </view>
  </view>
</template>

<script>
export default {
  name: "tki-tree",
  props: {
    lazy: {
      type: Boolean,
      default: false
    },
    range: {
      type: Array,
      default: function () {
        return []
      }
    },
    idKey: {
      type: String,
      default: 'id'
    },
    rangeKey: {
      type: String,
      default: 'label'
    },
    title: {
      type: String,
      default: ''
    },
    multiple: { // 是否可以多选
      type: Boolean,
      default: false
      // default: true
    },
    selectParent: { //是否可以选父级
      type: Boolean,
      default: false
    },
    foldAll: { //折叠时关闭所有已经打开的子集，再次打开时需要一级一级打开
      type: Boolean,
      default: false
    },
    confirmColor: { // 确定按钮颜色
      type: String,
      default: '' // #07bb07
    },
    cancelColor: { // 取消按钮颜色
      type: String,
      default: '' // #757575
    },
    titleColor: { // 标题颜色
      type: String,
      default: '' // #757575
    },
    currentIcon: { // 展开时候的ic
      type: String,
      default: ''
    },
    defaultIcon: { // 折叠时候的ic
      type: String,
      default: ''
    },
    lastIcon: { // 没有子集的ic
      type: String,
      default: ''
    },
    border: { // 是否有分割线
      type: Boolean,
      default: false
    },
  },
  data() {
    return {
      showTree: false,
      treeList: [],
      selectIndex: -1,
      returnedItem: [],//定义一个空数组
      pids: [],
      ancestorsIds: [],
      childNums: [],
    }
  },
  computed: {},
  methods: {
    _show() {
      this.showTree = true
    },
    _hide() {
      this.showTree = false
    },
    _cancel() {
      this._hide()
      this.$emit("cancel", '');
    },
    _confirm() {
      // 处理所选数据
      let rt = [],
          obj = {};
      this.treeList.forEach((v, i) => {
        if (this.treeList[i].checked) {
          rt.push(this.treeList[i].id)
        }
      })
      this._hide()
      this.$emit("confirm", rt);
    },
    //扁平化树结构
    _renderTreeList(list = [], rank = 0, parentId = [], parents = []) {
      list.forEach(item => {
        this.treeList.push({
          id: item[this.idKey],
          name: item[this.rangeKey],
          source: item,
          parentId, // 父级id数组
          parents, // 父级id数组
          rank, // 层级
          showChild: false, //子级是否显示
          open: false, //是否打开
          show: rank === 0, // 自身是否显示
          hideArr: [],
          orChecked: item.checked ? item.checked : false,
          checked: item.checked ? item.checked : false,
          childNum: 0
        })

        if (Array.isArray(item.children) && item.children.length > 0) {
          // console.log(item)
          let parentid = [...parentId],
              parentArr = [...parents];
          delete parentArr.children
          parentid.push(item[this.idKey]);
          parentArr.push({
            [this.idKey]: item[this.idKey],
            [this.rangeKey]: item[this.rangeKey]
          })
          // lazy
          if (!this.lazy) {
            this._renderTreeList(item.children, rank + 1, parentid, parentArr)
          }
        } else {
          this.treeList[this.treeList.length - 1].lastRank = true;
        }
      })
    },
    // 处理默认选择
    _defaultSelect() {
      this.treeList.forEach((v, i) => {
        if (v.checked) {
          this.treeList.forEach((v2, i2) => {
            if (v.parentId.toString().indexOf(v2.parentId.toString()) >= 0) {
              v2.show = true
              if (v.parentId.includes(v2.id)) {
                v2.showChild = true;
                v2.open = true;
              }
            }
          })
        }
      })
    },
    getOwn(id, arr) {
      //利用foreach循环遍历
      arr.forEach((item) => {
        //判断递归结束条件
        if (item[this.idKey] == id) {
          // 存储数据到空数组
          this.returnedItem = item
        } else if (item.children != null) //判断chlidren是否有数据
        {
          //递归调用
          this.getOwn(id, item.children);
        }
      })
      return this.returnedItem
    },
    setShow(id, arr, isShow) {
      arr.forEach((item, index) => {
        if (item.parentId.includes(id)) {
          this.treeList[index].showChild = isShow
          this.treeList[index].show = isShow
        } else if (item.children !== undefined) {
          this.setShow(id, item.children, isShow)
        }
      })
    },
    // 点击
    _treeItemTap(item, index) {
      // console.log(item)
      if (item.lastRank === true) {
        //点击最后一级时触发事件
        this.treeList[index].checked = !this.treeList[index].checked
        this._fixMultiple(index)
        return;
      }
      let id = item.id;
      item.showChild = !item.showChild;
      // qingqian
      if (item.showChild) {
        const range = this.range
        const parentIdArr = item.parentId
        // 找到当前元素
        const own = this.getOwn(id, range)
        const checkedChildren = own.children
        // 子元素插入的索引位置
        const nextIndex = this.treeList.findIndex(itemT => itemT.id === item.id)
        console.log(checkedChildren);
        if (checkedChildren === undefined || checkedChildren.length < 1) {
          return
        }
        // 子节点数量
        this.treeList[index].childNum = checkedChildren.length
        const newRank = item.rank + 1
        checkedChildren.forEach(itemC => {
          const childObj = {
            id: itemC[this.idKey],
            name: itemC[this.rangeKey],
            source: {},
            parentId: [item.id], // 父级id数组
            parents: [item], // 父级id数组
            rank: newRank, // 层级
            showChild: false, //子级是否显示
            open: false, //是否打开
            show: 1, // 自身是否显示
            hideArr: [],
            orChecked: this.treeList[index].checked,
            checked: this.treeList[index].checked,
          }
          if (!this.treeList.some(itemT => itemT.id === itemC[this.idKey])) {
            this.treeList.splice(nextIndex + 1, 0, childObj)
          }
        })
      }
      // 展开/隐藏子级/孙级
      let list = this.treeList
      item.open = item.showChild ? true : !item.open;
      list.forEach((childItem, i) => {
        if (item.showChild === false) {
          //隐藏所有子级
          if (!childItem.parentId.includes(id)) {
            return;
          }
          //TODO: 修改
          if (!this.foldAll) {
            if (childItem.lastRank !== true && !childItem.open) {
              childItem.showChild = false;
              this.setShow(childItem.id, this.treeList, false)
            }
            // 为隐藏的内容添加一个标记
            if (childItem.show) {
              childItem.hideArr[item.rank] = id
            }
          } else {
            if (childItem.lastRank !== true) {
              childItem.showChild = false;
              // 继续隐藏子级的的子级
              this.setShow(childItem.id, this.treeList, false)
            }
          }
          if (childItem.children !== undefined) {
            childItem.children.forEach((childItem1, i1) => {
              if (!childItem1.parentId.includes(childItem.id)) {
                return
              }
              childItem.children[i1].showChild = false
              childItem.children[i1].show = false
            })
          }
          childItem.show = false;
        } else {
          // 打开子集
          if (childItem.parentId[childItem.parentId.length - 1] === id) {
            childItem.show = true;
          }
          // 打开被隐藏的子集
          if (childItem.parentId.includes(id) && !this.foldAll) {
            // console.log(childItem.hideArr)
            if (childItem.hideArr[item.rank] === id) {
              childItem.show = true;
              if (childItem.open && childItem.showChild) {
                childItem.showChild = true
              } else {
                childItem.showChild = false
              }
              childItem.hideArr[item.rank] = null
            }
          }
        }
      })
    },
    // 通过父id处理子级
    syncChecked(trees, pid, checked) {
      trees.forEach((item, index) => {
        if (item.parentId.includes(pid)) {
          this.treeList[index].checked = checked
          this.syncChecked(trees, item.id, checked)
        } else if (item.children !== undefined) {
          this.syncChecked(item.children, pid, checked)
        }
      })
    },
    // 获取父级往上所有层级的id 并同步状态
    setAncestors(pids, checked) {
      this.treeList.forEach((item, index) => {
        if (pids.includes(item.id)) {
          if (checked && this.childNums[item.id] !== undefined && item.childNum === this.childNums[item.id]) {
            // 子级全部选中, 父级才选中
            this.treeList[index].checked = true
          } else {
            this.treeList[index].checked = false
          }
          this.setAncestors(item.parentId, checked)
        }
      })
    },
    _treeItemSelect(item, index) {
      this.treeList[index].checked = !this.treeList[index].checked
      // 选父级, 子级自动全选
      this.syncChecked(this.treeList, item.id, this.treeList[index].checked)

      if (item.rank > 0) {
        item.parentId.forEach((pid, indexP) => {
          const parent = this.treeList.filter(i => i.id === pid)
          const childNum = parent.length > 0 ? parent[0].childNum : 0
          if (this.childNums[pid] === undefined) {
            this.childNums[pid] = 1
          } else if (this.childNums[pid] < childNum) {
            this.childNums[pid]++
          }
        })
        //子级选择/选满/取消选择, 父级往上同步状态
        this.setAncestors(item.parentId, this.treeList[index].checked)
      }
      this._fixMultiple(index)
    },
    // 处理单选多选
    _fixMultiple(index) {
      if (!this.multiple) {
        // 如果是单选
        this.treeList.forEach((v, i) => {
          if (i != index) {
            this.treeList[i].checked = false
          } else {
            this.treeList[i].checked = true
          }
        })
      }
    },
    // 重置数据
    _reTreeList() {
      this.treeList.forEach((v, i) => {
        this.treeList[i].checked = v.orChecked
      })
    },
    _initTree(range = this.range) {
      this.treeList = [];
      this._renderTreeList(range);
      this.$nextTick(() => {
        this._defaultSelect(range)
      })
    }
  },
  watch: {
    range(list) {
      this._initTree(list);
    },
    multiple() {
      if (this.range.length) {
        this._reTreeList();
      }
    },
    selectParent() {
      if (this.range.length) {
        this._reTreeList();
      }
    },
  },
  mounted() {
    this._initTree();
  }
}
</script>

<style scoped>
@import "./style.css";
</style>
